#!/bin/bash

# 设置颜色变量
GREEN='\033[0;32m'
BLUE='\033[0;34m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# 打印带颜色的信息函数
print_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

print_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# 进度条函数（简化版，不依赖 bc）
show_progress() {
    echo -ne "${YELLOW}[进度] ${NC}"
    for ((i=0; i<20; i++)); do
        echo -ne "▓"
        sleep 0.1
    done
    echo
}

# 系统信息检查
print_info "系统信息检查中..."
echo "操作系统: $(cat /etc/redhat-release)"
echo "CPU核心数: $(nproc)"
echo "内存总量: $(free -h | awk '/Mem:/ {print $2}')"
echo "磁盘空间: $(df -h / | awk 'NR==2 {print $4}') (可用)"

# 安装必要的依赖包
print_info "安装系统依赖..."
yum install -y yum-utils device-mapper-persistent-data lvm2 curl wget &> /dev/null

# 配置 Docker 安装源
print_info "配置 Docker 安装源..."
yum install -y yum-utils &> /dev/null
yum-config-manager --add-repo http://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo &> /dev/null
yum makecache fast &> /dev/null

# 卸载旧版本 Docker（如果存在）
print_info "清理系统环境..."
yum remove -y docker \
    docker-client \
    docker-client-latest \
    docker-common \
    docker-latest \
    docker-latest-logrotate \
    docker-logrotate \
    docker-engine \
    docker-ce \
    docker-ce-cli \
    containerd.io &> /dev/null

# 检查 Docker 是否已安装并正常运行
check_docker() {
    if command -v docker &> /dev/null; then
        if docker info &> /dev/null; then
            print_success "Docker 已安装且正常运行"
            return 0
        else
            print_info "Docker 已安装但未运行，尝试启动服务..."
            systemctl start docker
            return 2
        fi
    fi
    return 1
}

# 在安装 Docker 之前先检查
if check_docker; then
    print_info "跳过 Docker 安装"
else
    print_info "开始安装 Docker..."
    # 安装 Docker CE
    yum install -y docker-ce docker-ce-cli containerd.io || {
        print_error "Docker 安装失败"
        exit 1
    }
    show_progress

    # 创建 Docker 配置目录
    mkdir -p /etc/docker

    # 配置 Docker 镜像加速
    print_info "配置 Docker 镜像加速..."
    cat > /etc/docker/daemon.json <<EOF
{
    "registry-mirrors": [
        "https://mirror.ccs.tencentyun.com",
        "https://hub-mirror.c.163.com",
        "https://docker.mirrors.ustc.edu.cn"
    ]
}
EOF

    # 启动 Docker
    print_info "启动 Docker 服务..."
    systemctl daemon-reload
    systemctl enable docker
    systemctl start docker || {
        print_error "Docker 服务启动失败"
        journalctl -xe --unit docker.service
        exit 1
    }
    show_progress

    # 等待 Docker 服务完全启动
    print_info "等待 Docker 服务就绪..."
    for i in {1..30}; do
        if docker info &> /dev/null; then
            break
        fi
        sleep 1
    done
fi

# 验证安装
if systemctl is-active docker &> /dev/null; then
    print_success "Docker服务运行状态: 正常"
    print_info "Docker Compose 现已集成在 Docker CLI 中，使用 'docker compose' 命令即可"
else
    print_error "Docker服务运行状态: 异常"
    journalctl -xe --unit docker.service
fi

echo -e "\n=== 安装完成 ==="
docker_version=$(docker --version 2>/dev/null)
echo "Docker 版本: ${docker_version:-未安装}"
echo "系统时间: $(date)"
echo "Docker 根目录: $(docker info 2>/dev/null | grep "Docker Root Dir" || echo "未知")"

print_info "推荐后续操作："
echo "1. 运行 'docker info' 查看详细信息"
echo "2. 运行 'docker run hello-world' 测试 Docker 安装"
echo "3. 使用 'docker compose version' 确认 Docker Compose 版本"

exit 0
